from selenium import webdriver
from selenium.webdriver.common.by import By
import time

# 初始化浏览器驱动
driver = webdriver.Chrome()

# 设置页面加载的超时时间（页面级等待）
driver.set_page_load_timeout(10)  # 单位：秒

try:
    # 访问一个网址
    driver.get("https://www.jd.com")
    print("页面加载成功")

    # 截图保存
    driver.save_screenshot("wait_page01.png")
    print("截图已保存")

except Exception as e:
    print(f"页面加载超时：{e}")

finally:
    # 关闭浏览器
    driver.quit()